import numpy as np

class Test:
    def __init__(self, n):
        self.data = np.arange(n)
        
    def __getitem__(self, id):
        return self.data[id]


if __name__ == "__main__":
    test = Test(10)
    print("test[7]", test[7])
